#!/usr/bin/env python
# encoding: utf-8
'''
@author: Excelsiorly
@license: (C) Copyright 2022, All Rights Reserved.
@contact: excelsiorly@qq.com
@file: 468. 验证IP地址.py
@time: 2022/1/29 16:33
@desc: https://leetcode-cn.com/problems/validate-ip-address/
> 给定一个字符串queryIP。如果是有效的 IPv4 地址，返回 "IPv4" ；如果是有效的 IPv6 地址，返回 "IPv6" ；如果不是上述类型的 IP 地址，返回 "Neither" 。
有效的IPv4地址 是 “x1.x2.x3.x4” 形式的IP地址。 其中0 <= xi<= 255且xi不能包含 前导零。例如:“192.168.1.1”、 “192.168.1.0” 为有效IPv4地址， “192.168.01.1” 为无效IPv4地址; “192.168.1.00” 、 “192.168@1.1” 为无效IPv4地址。
一个有效的IPv6地址是一个格式为“x1:x2:x3:x4:x5:x6:x7:x8” 的IP地址，其中:
1 <= xi.length <= 4
xi是一个 十六进制字符串 ，可以包含数字、小写英文字母( 'a' 到 'f' )和大写英文字母( 'A' 到 'F' )。
在xi中允许前导零。
例如 "2001:0db8:85a3:0000:0000:8a2e:0370:7334" 和 "2001:db8:85a3:0:0:8A2E:0370:7334" 是有效的 IPv6 地址，而 "2001:0db8:85a3::8A2E:037j:7334" 和 "02001:0db8:85a3:0000:0000:8a2e:0370:7334" 是无效的 IPv6 地址。

@解题思路：
    1. 硬做
'''
class Solution(object):
    def validIPAddress(self, queryIP):
        """
        :type queryIP: str
        :rtype: str
        """
        def validIPv4(queryIP):
            if not '.' in queryIP: return False
            segs = queryIP.split('.')
            if not len(segs)==4: return False
            for seg in segs:
                if len(seg)>1 and seg[0]=='0': return False
                else:
                    try:
                        seg = int(seg)
                    except:
                        return False
                    if not 0<=seg<=255:
                        return False
            return True

        def validIPv6(queryIP):
            if not ':' in queryIP: return False
            segs = queryIP.split(':')
            if not len(segs)==8: return False
            for seg in segs:
                if len(seg)>4 or not seg: return False
                for s in seg:
                    if not '0'<=s<='9' and not 'a'<=s<='f' and not 'A'<=s<='F':
                        return False
            return True
        if validIPv4(queryIP): return 'IPv4'
        if validIPv6(queryIP): return 'IPv6'
        return 'Neither'

if __name__ == '__main__':
    ip = "2001:db8:85a3:0::8a2E:0370:7334"
    res = Solution().validIPAddress(ip)
    print(res)